<template>
  <div class="app-container">
    <el-form :model="queryParams" ref="queryRef" :inline="true" v-show="showSearch" label-width="80px">
      <el-form-item label="流程名称" prop="name">
        <el-input
            v-model="queryParams.name"
            placeholder="请输入流程名称"
            clearable
            @keyup.enter="handleQuery"
        />
      </el-form-item>
      <el-form-item label="状态" prop="status">
        <el-select v-model="queryParams.status" style="width: 200px" placeholder="请选择状态" clearable>
          <el-option
              v-for="dict in ai_workflow_status"
              :key="dict.value"
              :label="dict.label"
              :value="dict.value"
          />
        </el-select>
      </el-form-item>
      <el-form-item>
        <el-button type="primary" icon="Search" @click="handleQuery">搜索</el-button>
        <el-button icon="Refresh" @click="resetQuery">重置</el-button>
      </el-form-item>
    </el-form>

    <el-row :gutter="10" class="mb8">
      <el-col :span="1.5">
        <el-button
            type="primary"
            plain
            icon="Plus"
            @click="handleAdd"
            v-hasPermi="['ai:aiWorkflow:add']"
        >新增
        </el-button>
      </el-col>
      <el-col :span="1.5">
        <el-button
            type="warning"
            plain
            icon="Download"
            @click="handleExport"
            v-hasPermi="['ai:aiWorkflow:export']"
        >导出
        </el-button>
      </el-col>
      <right-toolbar v-model:showSearch="showSearch" @queryTable="getList"></right-toolbar>
    </el-row>

    <el-table v-loading="loading" :data="aiWorkflowList" @selection-change="handleSelectionChange">
      <el-table-column type="selection" width="55" align="center"/>
      <el-table-column label="ID" align="center" prop="id"/>
      <el-table-column label="流程名称" align="center" prop="name"/>
      <el-table-column label="流程配置" align="center" prop="graph" show-overflow-tooltip/>
      <el-table-column label="状态" align="center" prop="status">
        <template #default="scope">
          <dict-tag :options="ai_workflow_status" :value="scope.row.status"/>
        </template>
      </el-table-column>
      <el-table-column label="备注" align="center" prop="remark" show-overflow-tooltip/>
      <el-table-column label="操作" align="center" class-name="small-padding fixed-width">
        <template #default="scope">
          <el-button link type="primary" icon="View" @click="handleDetail(scope.row)" v-show="scope.row.status === 1">流程查看</el-button>
          <el-button link type="primary" icon="EditPen" @click="handleDesign(scope.row)" v-hasPermi="['ai:aiWorkflow:design']"
                     v-show="scope.row.status === 0">流程设计
          </el-button>
          <el-button link type="primary" icon="Promotion" @click="handleRelease(scope.row)" v-hasPermi="['ai:aiWorkflow:release']"
                     v-show="scope.row.status === 0">发布
          </el-button>
          <el-button link type="primary" icon="Edit" @click="handleUpdate(scope.row)" v-hasPermi="['ai:aiWorkflow:edit']"
                     v-show="scope.row.status === 0">修改
          </el-button>
          <el-button link type="primary" icon="Delete" @click="handleDelete(scope.row)" v-hasPermi="['ai:aiWorkflow:remove']"
                     v-show="scope.row.status === 0">删除
          </el-button>
        </template>
      </el-table-column>
    </el-table>

    <pagination
        v-show="total>0"
        :total="total"
        v-model:page="queryParams.pageNum"
        v-model:limit="queryParams.pageSize"
        @pagination="getList"
    />

    <!-- 添加或修改AI流程对话框 -->
    <el-dialog :title="title" v-model="open" width="800px" append-to-body destroy-on-close>
      <el-form ref="aiWorkflowRef" :model="form" :rules="rules" label-width="120px">
        <el-form-item label="流程名称" prop="name">
          <el-input v-model="form.name" placeholder="请输入流程名称"/>
        </el-form-item>
        <el-form-item label="备注" prop="remark">
          <el-input v-model="form.remark" type="textarea" placeholder="请输入内容"/>
        </el-form-item>
      </el-form>
      <template #footer>
        <div class="dialog-footer">
          <el-button type="primary" @click="submitForm">确 定</el-button>
          <el-button @click="cancel">取 消</el-button>
        </div>
      </template>
    </el-dialog>

    <!--详情按钮 对话框-->
    <el-dialog :title="title" v-model="detailOpen" width="800px" append-to-body destroy-on-close>
      <el-form ref="aiWorkflowDetailRef" :model="form" :rules="rules" label-width="120px">
        <el-form-item label="流程名称" prop="name">
          <el-input v-model="form.name" placeholder="请输入流程名称" disabled/>
        </el-form-item>
        <el-form-item label="状态" prop="status">
          <el-radio-group v-model="form.status" disabled>
            <el-radio
                v-for="dict in ai_workflow_status"
                :key="dict.value"
                :label="parseInt(dict.value)"
            >{{ dict.label }}
            </el-radio>
          </el-radio-group>
        </el-form-item>
        <el-form-item label="备注" prop="remark">
          <el-input v-model="form.remark" type="textarea" placeholder="请输入内容" disabled/>
        </el-form-item>
      </el-form>
      <template #footer>
        <div class="dialog-footer">
          <el-button @click="cancel">关 闭</el-button>
        </div>
      </template>
    </el-dialog>

    <el-dialog :title="title" v-model="designOpen" fullscreen append-to-body destroy-on-close @close="handleCloseDesign">
      <WorkflowDesigner :flowItem="aiWorkFlowItem"/>
<!--      <WorkflowDesigner/>-->
    </el-dialog>
  </div>
</template>

<script setup name="AiWorkflow">
import {
  addAiWorkflow, canSaveFlag,
  delAiWorkflow, flowEdges,
  flowNodes,
  getAiWorkflow, knowledgeList,
  listAiWorkflow, llmModelList,
  releaseAiWorkflow,
  updateAiWorkflow
} from "@/api/ai/aiFlow.js";
import WorkflowDesigner from "@/views/ai/aiFlow/workflowDesigner.vue";
import {listAiEnableModel} from "@/api/ai/aiModel.js";
import {listAllCompleteKnowledge} from "@/api/ai/aiKnowledge.js";

const {proxy} = getCurrentInstance();
const {ai_workflow_status} = proxy.useDict('ai_workflow_status');

const aiWorkflowList = ref([]);
const open = ref(false);
const detailOpen = ref(false);
const designOpen = ref(false);
const loading = ref(true);
const showSearch = ref(true);
const ids = ref([]);
const single = ref(true);
const multiple = ref(true);
const total = ref(0);
const title = ref("");
const aiWorkFlowItem= ref({});

const data = reactive({
  form: {},
  queryParams: {
    pageNum: 1,
    pageSize: 10,
    name: null,
    graph: null,
    status: null,
  },
  rules: {
    name: [
      {required: true, message: "流程名称不能为空", trigger: "blur"}
    ],
    status: [
      {required: true, message: "状态不能为空", trigger: "change"}
    ],
  }
});

const {queryParams, form, rules} = toRefs(data);

function handleRelease(row) {
  // 发布前确认
  proxy.$modal.confirm('确认要发布"' + row.name + '"流程吗?').then(function () {
    releaseAiWorkflow(row.id).then(resp => {
      proxy.$modal.msgSuccess("发布成功")
      getList()
    })
  }).catch(function () {

  });
}

// 详情按钮操作
function handleDetail(row) {
  canSaveFlag.value = false
  if (row.graph !== null) {
    const graphJson = JSON.parse(row.graph)
    flowNodes.value  = graphJson.nodes
    flowEdges.value  = graphJson.edges
  }
  aiWorkFlowItem.value = row
  designOpen.value = true
}

function handleDesign(row) {
  if (row.graph !== null) {
    const graphJson = JSON.parse(row.graph)
    flowNodes.value  = graphJson.nodes
    flowEdges.value  = graphJson.edges
  }
  canSaveFlag.value = true
  aiWorkFlowItem.value = row
  designOpen.value = true
}

function handleCloseDesign() {
  designOpen.value = false
  flowNodes.value = []
  flowEdges.value = []
  aiWorkFlowItem.value = {}
  getList()
}


/** 查询AI流程列表 */
function getList() {
  loading.value = true;
  listAiWorkflow(queryParams.value).then(response => {
    aiWorkflowList.value = response.rows;
    total.value = response.total;
    loading.value = false;
  });
}

// 取消按钮
function cancel() {
  open.value = false;
  detailOpen.value = false;
  designOpen.value = false;
  reset();
}

// 表单重置
function reset() {
  form.value = {
    id: null,
    name: null,
    graph: null,
    status: null,
    createBy: null,
    createTime: null,
    updateBy: null,
    updateTime: null,
    remark: null
  };
  proxy.resetForm("aiWorkflowRef");
}

/** 搜索按钮操作 */
function handleQuery() {
  queryParams.value.pageNum = 1;
  getList();
}

/** 重置按钮操作 */
function resetQuery() {
  proxy.resetForm("queryRef");
  handleQuery();
}

// 多选框选中数据
function handleSelectionChange(selection) {
  ids.value = selection.map(item => item.id);
  single.value = selection.length != 1;
  multiple.value = !selection.length;
}

/** 新增按钮操作 */
function handleAdd() {
  reset();
  open.value = true;
  title.value = "添加AI流程";
}


/** 修改按钮操作 */
function handleUpdate(row) {
  reset();
  const _id = row.id || ids.value
  getAiWorkflow(_id).then(response => {
    form.value = response.data;
    open.value = true;
    title.value = "修改AI流程";
  });
}

/** 提交按钮 */
function submitForm() {
  proxy.$refs["aiWorkflowRef"].validate(valid => {
    if (valid) {
      if (form.value.id != null) {
        updateAiWorkflow(form.value).then(response => {
          proxy.$modal.msgSuccess("修改成功");
          open.value = false;
          getList();
        });
      } else {
        addAiWorkflow(form.value).then(response => {
          proxy.$modal.msgSuccess("新增成功");
          open.value = false;
          getList();
        });
      }
    }
  });
}

/** 删除按钮操作 */
function handleDelete(row) {
  const _ids = row.id || ids.value;
  proxy.$modal.confirm('是否确认删除AI流程编号为"' + _ids + '"的数据项？').then(function () {
    return delAiWorkflow(_ids);
  }).then(() => {
    getList();
    proxy.$modal.msgSuccess("删除成功");
  }).catch(() => {
  });
}

/** 导出按钮操作 */
function handleExport() {
  proxy.download('ai/aiWorkflow/export', {
    ...queryParams.value
  }, `aiWorkflow_${new Date().getTime()}.xlsx`)
}

function initLLMModel() {
  llmModelList.value = []
  listAiEnableModel().then(response => {
    response.data.forEach(item => {
      llmModelList.value.push({
        value: item.id,
        label: item.name,
      })
    })
  })
}

function initKnowledge() {
  knowledgeList.value = []
  listAllCompleteKnowledge().then(response => {
    response.data.forEach(item => {
      knowledgeList.value.push({
        value: item.id,
        label: item.name,
      })
    })
  })
}

initLLMModel();
initKnowledge();

getList();
</script>
